"""
main.py 系统主界面
"""

from model import FormattedMessage


class SysCli:
    
    def __init__(self):
        """
        定义格式化消息实体类对象
        """
        pass

    @staticmethod
    def main_menu():
        print('-------------------------')
        print('1.查询格式化消息')
        print('2.添加格式化消息')
        print('3.修改格式化消息')
        print('4.删除格式化消息')
        print('5.退出')

    @staticmethod
    def get_user_choice():
        try:
            c = int(input('请选择（1-5）：'))
            if 1 <= c <= 5:
                return c
            else:
                # 输入非法，将显示帮助信息
                return 0
        except ValueError:
            return 0
    
    def add_formatted_message(self):
        print('添加格式化消息功能')

    def get_formatted_message(self):
        print('查询格式化消息功能')

    def update_formatted_message(self):
        print('修改格式化消息功能')

    def del_formatted_message(self):
        print('删除格式化消息功能')

    def call_function(self, choice=0):
        if choice == 0:
            print('请选择正确的选项！')
        elif choice == 1: # 查询格式化消息
            self.get_formatted_message()
        elif choice == 2: # 添加格式化消息
            self.add_formatted_message()
        elif choice == 3: # 修改格式化消息
            self.update_formatted_message()
        elif choice == 4: # 删除格式化消息
            self.del_formatted_message()
        elif choice == 5: # 退出系统
            print('退出系统')
            exit(0)

    def run(self):
        while True:
            self.main_menu()
            t = self.get_user_choice()
            self.call_function(t)

MsgSystem = SysCli()
MsgSystem.run()